from rest_framework.pagination import PageNumberPagination
from rest_framework.response import Response


def jwt_response_payload_handler(token, user=None, request=None):
    return {
        'token': token,
        'id': user.id,
        'username': user.username,

    }


class PageNum(PageNumberPagination):
    """
    定义分页器
    """
    # 可以在子类中定义的属性：
    # page_size  # 每页数目
    # page_query_para # 前端发送的页数关键字名，默认为 "page"
    # page_size_query_param # 前端发送的每页数目关键字名，默认为None
    # max_page_size   # 前端最多能设置的每页数量
    page_size = 5  # 在没有指定每页返回数量的情况下，后端指定返回五条数据
    page_size_query_param = 'pagesize'  # 指定控制每页显示数量的参数
    max_page_size = 10

    # 改写分页结果返回方法
    def get_paginated_response(self, data):
        return Response(
            {
                'count': self.page.paginator.count,
                'lists': data,
                'page': self.page.number,  # 获取当前页码
                'pages': self.page.paginator.num_pages,
                'pagesize': self.page_size
            }
        )
